#include "BlockQueue.hpp"
#include <iostream>
#include <pthread.h>
#include <unistd.h>
#include "Task.hpp"

void *consumer(void *args)
{
    BlockQueue<Task> *bq = static_cast<BlockQueue<Task> *>(args);

    // 消费者不断消费
    while (true)
    {
        sleep(1);
        // 消费者消费数据
        Task t = bq->Pop();

        t.Execute();
        std::cout << "消费了一个任务：" << t.X() << "+" << t.Y() << "=" << t.Result() <<  std::endl;
    }
}

void *productor(void *args)
{
    int x = 1, y = 1;
    BlockQueue<Task> *bq = static_cast<BlockQueue<Task> *>(args);
    // 生产者不断生产
    while (true)
    {
        // 生产者向队列里派发任务
        std::cout << "生产了一个任务" <<x << "+"<< y << "= ?"<<  std::endl;
        Task t(x,y);
        bq->Equeue(t);
        x++, y++;
    }
}

//这个代码既支持单生产，单消费
//也支持多生产，多消费
int main()
{
    // 申请队列空间
    BlockQueue<Task> *bq = new BlockQueue<Task>();

    // 构建生产和消费
    pthread_t c[2], p[3];
    pthread_create(c, nullptr, consumer, bq);
    pthread_create(c+1, nullptr, consumer, bq);

    pthread_create(p, nullptr, productor, bq);
    pthread_create(p+1, nullptr, productor, bq);
    pthread_create(p+2, nullptr, productor, bq);


    pthread_join(c[0], nullptr);
    pthread_join(c[1], nullptr);

    pthread_join(p[0], nullptr);
    pthread_join(p[1], nullptr);
    pthread_join(p[2], nullptr);


    return 0;
}